import vue from '@vitejs/plugin-vue';
import {resolve} from 'path';
import {ConfigEnv, defineConfig, loadEnv} from 'vite';
import vueSetupExtend from 'vite-plugin-vue-setup-extend';
import AutoImport from 'unplugin-auto-import/vite';
import topLevelAwait from 'vite-plugin-top-level-await';
import {createStyleImportPlugin, VxeTableResolve} from 'vite-plugin-style-import';
import viteCompression from 'vite-plugin-compression';
// @ts-ignore
import {svgBuilder} from '/@/components/IconSelector/index';

const pathResolve = (dir: string) => {
    return resolve(__dirname, '.', dir);
};

const alias: Record<string, string> = {
    '/@': pathResolve('./src/'),
    'vue-i18n': 'vue-i18n/dist/vue-i18n.cjs.js',
};

const viteConfig = defineConfig((mode: ConfigEnv) => {
    const env = loadEnv(mode.mode, process.cwd());
    return {
        plugins: [
            vue(), // Vue 插件
            svgBuilder('./src/assets/icons/'), // 将 SVG 文件转换成 Vue 组件
            vueSetupExtend(), // setup语法糖增强插件
            AutoImport({
                imports: ['vue', 'vue-router', 'pinia'], // 自动导入的依赖库数组
                dts: './auto-imports.d.ts', // 自动导入类型定义文件路径
            }),
            createStyleImportPlugin({
                resolves: [VxeTableResolve()], // 配置vxetable 按需加载
            }),
            topLevelAwait({
                promiseExportName: '__tla', // TLA Promise 变量名
                promiseImportName: (i) => `__tla_${i}`, // TLA Promise 导入名
            }),
            viteCompression({
                deleteOriginFile: false, // 压缩后删除原来的文件
            }),
        ],
        root: process.cwd(), // 项目根目录
        resolve: {alias}, // 路径别名配置
        base: mode.command === 'serve' ? './' : env.VITE_PUBLIC_PATH,
        optimizeDeps: {
            include: ['element-plus/es/locale/lang/zh-cn', 'element-plus/es/locale/lang/en'],
        },
        server: {
            host: '0.0.0.0', // 服务器地址
            port: env.VITE_PORT as unknown as number, // 服务器端口号
            open: env.VITE_OPEN === 'true', // 是否自动打开浏览器
            hmr: true, // 启用热更新
            proxy: {
                '/api/gen': {
                    //单体架构下特殊处理代码生成模块代理
                    target: env.VITE_IS_MICRO === 'true' ? env.VITE_ADMIN_PROXY_PATH : env.VITE_GEN_PROXY_PATH,
                    changeOrigin: true,
                    rewrite: (path) => path.replace(/^\/api/, ''),
                },
                '/api/wms': {
                    target: env.VITE_WMS_PROXY_PATH,
                    changeOrigin: true,
                    rewrite: (path) => path.replace(/^\/api/, ''),
                    bypass: (req, res, options) => {
                        // 去掉Bearer
                        req.headers["authorization"] = req.headers["authorization"]?.replace(/^Bearer /, '')
                        // 记录真实请求的地址
                        req.headers['X-Real-Host'] = options.target + options.rewrite(req.url)
                        res.setHeader('X-Real-Host', options.target + options.rewrite(req.url))
                    }
                },
                '/api/report': {
                    target: env.VITE_REPORT_PROXY_PATH,
                    changeOrigin: true,
                    rewrite: (path) => path.replace(/^\/api/, ''),
                    bypass: (req, res, options) => {
                        // 去掉Bearer
                        req.headers["authorization"] = req.headers["authorization"]?.replace(/^Bearer /, '')
                        // 记录真实请求的地址
                        req.headers['X-Real-Host'] = options.target + options.rewrite(req.url)
                        res.setHeader('X-Real-Host', options.target + options.rewrite(req.url))
                    }
                },
                '/api/iot': {
                    target: env.VITE_IOT_PROXY_PATH,
                    changeOrigin: true,
                    rewrite: (path) => path.replace(/^\/api/, ''),
                    bypass: (req, res, options) => {
                        // 去掉Bearer
                        req.headers["authorization"] = req.headers["authorization"]?.replace(/^Bearer /, '')
                        // 记录真实请求的地址
                        req.headers['X-Real-Host'] = options.target + options.rewrite(req.url)
                        res.setHeader('X-Real-Host', options.target + options.rewrite(req.url))
                    }
                },
                '/api': {
                    target: env.VITE_ADMIN_PROXY_PATH, // 目标服务器地址
                    ws: true, // 是否启用 WebSocket
                    changeOrigin: true, // 是否修改请求头中的 Origin 字段
                    rewrite: (path) => path.replace(/^\/api/, ''),
                    bypass: (req, res, options) => {
                        // 去掉Bearer
                        // 记录真实请求的地址
                        req.headers['X-Real-Host'] = options.target + options.rewrite(req.url)
                        res.setHeader('X-Real-Host', options.target + options.rewrite(req.url))
                    }
                },
                '^/ws/info/.*': {
                    target: env.VITE_ADMIN_PROXY_PATH, // 目标服务器地址
                    ws: true, // 是否启用 WebSocket
                    changeOrigin: true,
                },
            },
        },
        build: {
            outDir: 'dist', // 打包输出目录
            chunkSizeWarningLimit: 1500, // 代码分包阈值
            rollupOptions: {
                output: {
                    entryFileNames: `assets/[name].[hash].js`,
                    chunkFileNames: `assets/[name].[hash].js`,
                    assetFileNames: `assets/[name].[hash].[ext]`,
                    compact: true,
                    manualChunks: {
                        vue: ['vue', 'vue-router', 'pinia'],
                        echarts: ['echarts'],
                    },
                },
            },
        },
        css: {preprocessorOptions: {css: {charset: false}}},
        define: {
            __VUE_I18N_LEGACY_API__: JSON.stringify(false),
            __VUE_I18N_FULL_INSTALL__: JSON.stringify(false),
            __INTLIFY_PROD_DEVTOOLS__: JSON.stringify(false),
            __VERSION__: JSON.stringify(process.env.npm_package_version),
            __NEXT_NAME__: JSON.stringify(process.env.npm_package_name),
        },
    };
});

export default viteConfig;
